import os
from openpyxl import Workbook, load_workbook
from datetime import datetime

data = [['小餐饮经营登记（延续）', '小餐饮经营登记', 'FAD2301BE5B940CD9C71E4007BEA7D13']]


def write(data):
    # 配置参数
    headers = ['事项名', '目录名', '标准编码']
    column_width = 50
    current_date = datetime.now().strftime("%Y-%m-%d")
    file_name = f"{current_date}.xlsx"  # 以当前日期为文件名

    # 检查文件是否存在
    file_exists = os.path.exists(file_name)

    if not file_exists:
        # 1. 文件不存在：创建新文件并写入表头和数据
        wb = Workbook()
        ws = wb.active
        ws.title = "数据表格"

        # 写入表头
        for col, header in enumerate(headers, start=1):
            ws.cell(row=1, column=col, value=header)

        # 设置列宽（仅新文件需要初始化）
        columns = ['A', 'B', 'C']
        for col in columns:
            ws.column_dimensions[col].width = column_width

        # 写入数据（从第2行开始）
        start_row = 2

    else:
        # 2. 文件已存在：加载文件并在最后一行后追加数据
        wb = load_workbook(file_name)
        ws = wb.active  # 获取当前活跃工作表

        # 找到最后一行的行号（下一行作为数据起始行）
        start_row = ws.max_row + 1

    # 追加数据（无论文件是否存在，都从start_row开始写入）
    for row_idx, row_data in enumerate(data, start=start_row):
        for col_idx, value in enumerate(row_data, start=1):
            ws.cell(row=row_idx, column=col_idx, value=value)

    # 保存文件
    wb.save(file_name)
    print(f"数据已成功{'追加' if file_exists else '写入'}！文件名：{file_name}，路径：当前工作目录")


# 第一次调用：创建文件并写入数据
# write(data)

# 测试追加功能（取消注释可测试）
# new_data = [['食品经营许可（新办）', '食品经营许可', '1234567890ABCDEF']]
# write(new_data)
